from pathlib import Path

# 项目根目录
ROOT_DIR = Path(__file__).parent.parent
# 数据路径
RAW_DATA_DIR = ROOT_DIR / 'data' / 'raw'
PROCESSED_DATA_DIR = ROOT_DIR / 'data' / 'processed'
# 模型和日志路径
MODELS_DIR = ROOT_DIR / 'models'
LOGS_DIR = ROOT_DIR / 'logs'

# 预训练模型路径
PRETRAINED_DIR = ROOT_DIR / 'pretrained'

# 训练参数
SEQ_LEN = 128  # 输入序列长度
BATCH_SIZE = 64  # 批次大小
LEARNING_RATE = 1e-3  # 学习率
EPOCHS = 3  # 训练轮数

RANDOM_SEED = 42
